package at.ac.tuwien.isis.infret.ue3.testing;

import static org.junit.Assert.*;

import java.io.File;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import at.ac.tuwien.isis.infret.helper.DocumentContainer;
import at.ac.tuwien.isis.infret.ue3.ArffParser;

public class ArffParserTest {


	private ArffParser test;
	private String[] docNames = {"classical.00094.mp3", "classical.00006.mp3",  "classical.00040.mp3"};
	@Before
	public void setUp() throws Exception {
		test = new ArffParser(
				"res"+File.separatorChar+
				"testing"+File.separatorChar+
				"testing.rh.arff");
	}

	@After
	public void tearDown() throws Exception {
	}

	@Test
	public void testGetDocumentContainer() {
		int i = 0;
		
		for(DocumentContainer doc : test.getDocumentContainer()){
			//System.out.println(doc.getDocumentname());
			assertEquals(docNames[i],doc.getDocumentname());
			assertEquals("classical",doc.getClassname());
			i++;
		}
	}

}
